from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq, AutoTokenizer, AutoModelForCausalLM
from models.ai.config.settings import MODEL_PATH

models = [
    "openai/whisper-large-v3"
]

for model_name in models:
    processor = AutoProcessor.from_pretrained(model_name, cache_dir=MODEL_PATH)
    model = AutoModelForSpeechSeq2Seq.from_pretrained(model_name, cache_dir=MODEL_PATH)

    # tokenizer = AutoTokenizer.from_pretrained(model_name, cache_dir=model_path)
    # model = AutoModelForCausalLM.from_pretrained(model_name, cache_dir=model_path)